/*
 * Copyright (C) 
 * Copyright (C) 
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

/* ScriptData
SDName: Desolace
SD%Complete: 100
SDComment: Quest support: 5561
SDCategory: Desolace
EndScriptData */

/* ContentData
npc_aged_dying_ancient_kodo
npc_dalinda_malem
go_demon_portal
EndContentData */

#include "ScriptMgr.h"
#include "ScriptedCreature.h"
#include "ScriptedGossip.h"
#include "ScriptedEscortAI.h"
#include "Player.h"
#include "SpellInfo.h"

// Ours
enum Caravan
{
    QUEST_BODYGUARD_FOR_HIRE			= 5821,
	QUEST_GIZELTON_CARAVAN				= 5943,

	EVENT_RESUME_PATH					= 1,
	EVENT_WAIT_FOR_ASSIST				= 2,
	EVENT_RESTART_ESCORT				= 3,

	NPC_CORK_GIZELTON					= 11625,
	NPC_RIGGER_GIZELTON					= 11626,
	NPC_CARAVAN_KODO					= 11564,
	NPC_VENDOR_TRON						= 12245,
	NPC_SUPER_SELLER					= 12246,

	SAY_CARAVAN_LEAVE					= 0,
	SAY_CARAVAN_HIRE					= 1,

	MAX_CARAVAN_SUMMONS					= 3,

	TIME_SHOP_STOP						= 10*MINUTE*IN_MILLISECONDS,
	TIME_HIRE_STOP						= 4*MINUTE*IN_MILLISECONDS,

	// Ambush
	NPC_KOLKAR_WAYLAYER					= 12976,
	NPC_KOLKAR_AMBUSHER					= 12977,
	NPC_LESSER_INFERNAL					= 4676,
	NPC_DOOMWARDER						= 4677,
	NPC_NETHER							= 4684,

};

class npc_cork_gizelton : public CreatureScript
{
	public:
		npc_cork_gizelton() : CreatureScript("npc_cork_gizelton") { }

		bool OnQuestAccept(Player* player, Creature* creature, Quest const* quest)
		{
			if (quest->GetQuestId() == QUEST_BODYGUARD_FOR_HIRE)
				creature->AI()->SetGUID(player->GetGUID(), player->getFaction());

			return true;
		}

		CreatureAI* GetAI(Creature* creature) const
		{
			return new npc_cork_gizeltonAI(creature);
		}

		struct npc_cork_gizeltonAI : public npc_escortAI
		{
			npc_cork_gizeltonAI(Creature* creature) : npc_escortAI(creature)
			{
				memset(&summons, 0, sizeof(summons));
			}

			EventMap events;
			uint64 summons[MAX_CARAVAN_SUMMONS];
			bool headNorth;

			uint64 _playerGUID;
			uint32 _faction;

			void Initialize()
			{
				_playerGUID = 0;
				_faction = 35;
				headNorth = true;
				me->setActive(true);
				events.ScheduleEvent(EVENT_RESTART_ESCORT, 0);
			}

			void JustRespawned()
			{
				npc_escortAI::JustRespawned();
				Initialize();
			}

			void InitializeAI()
			{
				npc_escortAI::InitializeAI();
				Initialize();
			}

			void JustDied(Unit* killer)
			{
				RemoveSummons();
				npc_escortAI::JustDied(killer);
			}

			void EnterEvadeMode()
			{
				SummonsFollow();
				ImmuneFlagSet(false, 35);
				npc_escortAI::EnterEvadeMode();
			}

			void CheckPlayer()
			{
				if (_playerGUID)
					if (Player* player = ObjectAccessor::GetPlayer(*me, _playerGUID))
						if (me->IsWithinDist(player, 60.0f))
							return;

				_playerGUID = 0;
				_faction = 35;
				ImmuneFlagSet(false, _faction);
			}

			void SetGUID(uint64 playerGUID, int32 faction)
			{
				_playerGUID = playerGUID;
				_faction = faction;
				SetEscortPaused(false);
				if (Creature* active = !headNorth ? me : ObjectAccessor::GetCreature(*me, summons[0]))
					active->RemoveFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);
				events.CancelEvent(EVENT_WAIT_FOR_ASSIST);
			}

			void SetData(uint32 field, uint32 data)
			{
				if (field == 1 && data == 1)
					if (Player* player = me->SelectNearestPlayer(50.0f))
						SetGUID(player->GetGUID(), player->getFaction());
			}

			bool CheckCaravan()
			{
				for (uint8 i = 0; i < MAX_CARAVAN_SUMMONS; ++i)
				{
					if (summons[i] == 0)
					{
						SummonHelpers();
						return false;
					}

					Creature* summon = ObjectAccessor::GetCreature(*me, summons[i]);
					if (!summon || me->GetDistance2d(summon) > 25.0f)
					{
						SummonHelpers();
						return false;
					}
				}
				return true;
			}

			void RemoveSummons()
			{
				for (uint8 i = 0; i < MAX_CARAVAN_SUMMONS; ++i)
				{
					if (Creature* summon = ObjectAccessor::GetCreature(*me, summons[i]))
						summon->DespawnOrUnsummon();
					
					summons[i] = 0;
				}
			}

			void SummonHelpers()
			{
				RemoveSummons();
				me->RemoveFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);

				Creature* cr = NULL;
				if (cr = me->SummonCreature(NPC_RIGGER_GIZELTON, *me))
				{
					cr->RemoveFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);
					summons[0] = cr->GetGUID();
				}
				if (cr = me->SummonCreature(NPC_CARAVAN_KODO, *me))
					summons[1] = cr->GetGUID();
				if (cr = me->SummonCreature(NPC_CARAVAN_KODO, *me))
					summons[2] = cr->GetGUID();
				
				SummonsFollow();
			}

			void SummonedCreatureDies(Creature* creature, Unit*)
			{
				if (creature->GetGUID() == summons[0])
					summons[0] = 0;
				else if (creature->GetGUID() == summons[1])
					summons[1] = 0;
				else if (creature->GetGUID() == summons[2])
					summons[2] = 0;
			}

			void SummonedCreatureDespawn(Creature* creature)
			{
				if (creature->GetGUID() == summons[0])
					summons[0] = 0;
				else if (creature->GetGUID() == summons[1])
					summons[1] = 0;
				else if (creature->GetGUID() == summons[2])
					summons[2] = 0;
			}

			void SummonsFollow()
			{
				float dist = 1.0f;
				for (uint8 i = 0; i < MAX_CARAVAN_SUMMONS; ++i)
					if (Creature* summon = ObjectAccessor::GetCreature(*me, summons[i]))
					{
						summon->GetMotionMaster()->Clear(false);
						summon->StopMoving();
						summon->GetMotionMaster()->MoveFollow(me, dist, M_PI, MOTION_SLOT_ACTIVE);
						dist += (i == 1 ? 9.5f : 3.0f);
					}
			}

			void RelocateSummons()
			{
				for (uint8 i = 0; i < MAX_CARAVAN_SUMMONS; ++i)
					if (Creature* summon = ObjectAccessor::GetCreature(*me, summons[i]))
						summon->SetHomePosition(me->GetPositionX(), me->GetPositionY(), me->GetPositionZ(), me->GetOrientation());
			}

			void ImmuneFlagSet(bool remove, uint32 faction)
			{
				for (uint8 i = 0; i < MAX_CARAVAN_SUMMONS; ++i)
					if (Creature* summon = ObjectAccessor::GetCreature(*me, summons[i]))
					{
						summon->setFaction(faction);
						if (remove)
							summon->RemoveFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_IMMUNE_TO_NPC);
						else
							summon->SetFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_IMMUNE_TO_NPC);
					}
				if (remove)
					me->RemoveFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_IMMUNE_TO_NPC);
				else
					me->SetFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_IMMUNE_TO_NPC);
				me->setFaction(faction);
			}

			void WaypointReached(uint32 waypointId)
			{
				RelocateSummons();
				switch (waypointId)
				{
					// Finished north path
					case 52:
						me->SummonCreature(NPC_VENDOR_TRON, -694.61f, 1460.7f, 90.794f, 2.4f, TEMPSUMMON_TIMED_DESPAWN, TIME_SHOP_STOP+15*IN_MILLISECONDS);
						SetEscortPaused(true);
						events.ScheduleEvent(EVENT_RESUME_PATH, TIME_SHOP_STOP);
						CheckCaravan();
						break;
					// Finished south path
					case 193:
						me->SummonCreature(NPC_SUPER_SELLER, -1905.5f, 2463.3f, 61.52f, 5.87f, TEMPSUMMON_TIMED_DESPAWN, TIME_SHOP_STOP+15*IN_MILLISECONDS);
						SetEscortPaused(true);
						events.ScheduleEvent(EVENT_RESUME_PATH, TIME_SHOP_STOP);
						CheckCaravan();
						break;
					// North -> South - hire
					case 77:
						SetEscortPaused(true);
						me->SetFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);
						Talk(SAY_CARAVAN_HIRE);
						events.ScheduleEvent(EVENT_WAIT_FOR_ASSIST, TIME_HIRE_STOP);
						break;
					// Sout -> North - hire
					case 208:
						SetEscortPaused(true);
						if (Creature* rigger = ObjectAccessor::GetCreature(*me, summons[0]))
						{
							rigger->SetFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);
							rigger->AI()->Talk(SAY_CARAVAN_HIRE);
						}
						events.ScheduleEvent(EVENT_WAIT_FOR_ASSIST, TIME_HIRE_STOP);
						break;
					// North -> South - complete
					case 103:
						if (Player* player = ObjectAccessor::GetPlayer(*me, _playerGUID))
						{
							if (CheckCaravan())
								player->GroupEventHappens(QUEST_BODYGUARD_FOR_HIRE, player);
							else
								player->FailQuest(QUEST_BODYGUARD_FOR_HIRE);
						}
						_playerGUID = 0;
						CheckPlayer();
						break;
					// South -> North - complete
					case 235:
						if (Player* player = ObjectAccessor::GetPlayer(*me, _playerGUID))
						{
							if (CheckCaravan())
								player->GroupEventHappens(QUEST_GIZELTON_CARAVAN, player);
							else
								player->FailQuest(QUEST_GIZELTON_CARAVAN);
						}
						_playerGUID = 0;
						CheckPlayer();
						break;
					// North -> South - spawn attackers
					case 83:
					case 93:
					case 100:
					{
						if (!_playerGUID)
							return;
						ImmuneFlagSet(true, _faction);
						Creature* cr = NULL;
						for (uint8 i = 0; i < 4; ++i)
						{
							float o = (i*M_PI/2)+(M_PI/4);
							float x = me->GetPositionX()+cos(o)*15.0f;
							float y = me->GetPositionY()+sin(o)*15.0f;
							if (cr = me->SummonCreature((i%2 == 0 ? NPC_KOLKAR_WAYLAYER : NPC_KOLKAR_AMBUSHER),
								x, y, me->GetMap()->GetHeight(x, y, MAX_HEIGHT), 0.0f, TEMPSUMMON_TIMED_DESPAWN_OUT_OF_COMBAT, 30000))
								cr->AI()->AttackStart(me);
						}
						if (cr)
						{
							AttackStart(cr);
							me->CallForHelp(50.0f);
						}
						break;
					}
					// South -> North - spawn attackers
					case 221:
					case 228:
					case 233:
					{
						if (!_playerGUID)
							return;
						ImmuneFlagSet(true, _faction);
						Creature* cr = NULL;
						for (uint8 i = 0; i < 3; ++i)
						{
							float o = i*2*M_PI/3;
							float x = me->GetPositionX()+cos(o)*10.0f;
							float y = me->GetPositionY()+sin(o)*10.0f;
							uint32 entry = NPC_LESSER_INFERNAL;
							if (i)
								entry = i == 1 ? NPC_DOOMWARDER : NPC_NETHER;

							if (cr = me->SummonCreature(entry, x, y, me->GetMap()->GetHeight(x, y, MAX_HEIGHT), 0.0f, TEMPSUMMON_TIMED_DESPAWN_OUT_OF_COMBAT, 30000))
								cr->AI()->AttackStart(me);
						}
						if (cr)
						{
							AttackStart(cr);
							me->CallForHelp(50.0f);
						}
						break;
					}
					case 282:
						events.ScheduleEvent(EVENT_RESTART_ESCORT, 1000);
						break;

				}
			}

			void UpdateEscortAI(uint32 diff)
			{
				events.Update(diff);
				switch (events.ExecuteEvent())
				{
					case EVENT_RESUME_PATH:
						SetEscortPaused(false);
						if (Creature* talker = headNorth ? me : ObjectAccessor::GetCreature(*me, summons[0]))
							talker->AI()->Talk(SAY_CARAVAN_LEAVE);

						headNorth = !headNorth;
						break;
					case EVENT_WAIT_FOR_ASSIST:
						SetEscortPaused(false);
						if (Creature* active = !headNorth ? me : ObjectAccessor::GetCreature(*me, summons[0]))
							active->RemoveFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_QUESTGIVER);
						break;
					case EVENT_RESTART_ESCORT:
						CheckCaravan();
						SetDespawnAtEnd(false);
						Start(true, true, 0, 0, false, false, true);
						break;
				}

				if (!UpdateVictim())
					return;

				DoMeleeAttackIfReady();
			}
		};
};


// Theirs
enum DyingKodo
{
    SAY_SMEED_HOME                  = 0,

    QUEST_KODO                      = 5561,

    NPC_SMEED                       = 11596,
    NPC_AGED_KODO                   = 4700,
    NPC_DYING_KODO                  = 4701,
    NPC_ANCIENT_KODO                = 4702,
    NPC_TAMED_KODO                  = 11627,

    SPELL_KODO_KOMBO_ITEM           = 18153,
    SPELL_KODO_KOMBO_PLAYER_BUFF    = 18172,
    SPELL_KODO_KOMBO_DESPAWN_BUFF   = 18377,
    SPELL_KODO_KOMBO_GOSSIP         = 18362

};

class npc_aged_dying_ancient_kodo : public CreatureScript
{
public:
    npc_aged_dying_ancient_kodo() : CreatureScript("npc_aged_dying_ancient_kodo") { }

    struct npc_aged_dying_ancient_kodoAI : public ScriptedAI
    {
        npc_aged_dying_ancient_kodoAI(Creature* creature) : ScriptedAI(creature) {}

		void JustRespawned()
		{
			me->UpdateEntry(RAND(NPC_AGED_KODO, NPC_DYING_KODO, NPC_ANCIENT_KODO), NULL, false);
		}

        void MoveInLineOfSight(Unit* who)
        {
            if (who->GetEntry() == NPC_SMEED && me->IsWithinDistInMap(who, 10.0f) && !me->HasAura(SPELL_KODO_KOMBO_GOSSIP))
            {
				me->SetHomePosition(me->GetPositionX(), me->GetPositionY(), me->GetPositionZ(), me->GetOrientation());
                me->GetMotionMaster()->Clear();

                DoCast(me, SPELL_KODO_KOMBO_GOSSIP, true);
                if (Creature* smeed = who->ToCreature())
                    smeed->AI()->Talk(SAY_SMEED_HOME);
            }
        }

        void SpellHit(Unit* caster, SpellInfo const* spell)
        {
            if (spell->Id == SPELL_KODO_KOMBO_ITEM)
            {
                if (!(caster->HasAura(SPELL_KODO_KOMBO_PLAYER_BUFF) || me->HasAura(SPELL_KODO_KOMBO_DESPAWN_BUFF))
                    && (me->GetEntry() == NPC_AGED_KODO || me->GetEntry() == NPC_DYING_KODO || me->GetEntry() == NPC_ANCIENT_KODO))
                {
                    me->UpdateEntry(NPC_TAMED_KODO, NULL, false);
					EnterEvadeMode();
                    me->GetMotionMaster()->MoveFollow(caster, PET_FOLLOW_DIST, me->GetFollowAngle());

					caster->CastSpell(caster, SPELL_KODO_KOMBO_PLAYER_BUFF, true);
                    DoCast(me, SPELL_KODO_KOMBO_DESPAWN_BUFF, true);
                }
            }
            else if (spell->Id == SPELL_KODO_KOMBO_GOSSIP)
            {
                me->SetFlag(UNIT_NPC_FLAGS, UNIT_NPC_FLAG_GOSSIP);
                me->DespawnOrUnsummon(60000);
            }
        }
    };

    bool OnGossipHello(Player* player, Creature* creature)
    {
        if (player->HasAura(SPELL_KODO_KOMBO_PLAYER_BUFF) && creature->HasAura(SPELL_KODO_KOMBO_DESPAWN_BUFF))
        {
            player->TalkedToCreature(creature->GetEntry(), 0);
            player->RemoveAurasDueToSpell(SPELL_KODO_KOMBO_PLAYER_BUFF);
        }

        player->SEND_GOSSIP_MENU(player->GetGossipTextId(creature), creature->GetGUID());
        return true;
    }

    CreatureAI* GetAI(Creature* creature) const
    {
        return new npc_aged_dying_ancient_kodoAI(creature);
    }

};

/*######
## npc_dalinda_malem. Quest 1440
######*/

enum Dalinda
{
    QUEST_RETURN_TO_VAHLARRIEL      = 1440
};

class npc_dalinda : public CreatureScript
{
public:
    npc_dalinda() : CreatureScript("npc_dalinda") { }

    struct npc_dalindaAI : public npc_escortAI
    {
        npc_dalindaAI(Creature* creature) : npc_escortAI(creature) { }

        void Reset() { }

        void EnterCombat(Unit* /*who*/) { }

        void JustDied(Unit* /*killer*/)
        {
            if (Player* player = GetPlayerForEscort())
                player->FailQuest(QUEST_RETURN_TO_VAHLARRIEL);
            return;
        }

        void WaypointReached(uint32 waypointId)
        {
            Player* player = GetPlayerForEscort();

            switch (waypointId)
            {
                case 1:
                    me->SetStandState(UNIT_STAND_STATE_STAND);
                    break;
                case 15:
                    if (player)
                        player->GroupEventHappens(QUEST_RETURN_TO_VAHLARRIEL, me);
                    break;
            }
        }

        void UpdateAI(uint32 diff)
        {
            npc_escortAI::UpdateAI(diff);

            if (!UpdateVictim())
                return;

            DoMeleeAttackIfReady();
        }
    };

    bool OnQuestAccept(Player* player, Creature* creature, Quest const* quest)
    {
        if (quest->GetQuestId() == QUEST_RETURN_TO_VAHLARRIEL)
       {
            if (npc_escortAI* escortAI = CAST_AI(npc_dalinda::npc_dalindaAI, creature->AI()))
            {
                escortAI->Start(true, false, player->GetGUID());
                creature->setFaction(113);
            }
        }
        return true;
    }

    CreatureAI* GetAI(Creature* creature) const
    {
        return new npc_dalindaAI(creature);
    }
};

/*######
## go_demon_portal
######*/

enum DemonPortal
{
    NPC_DEMON_GUARDIAN          = 11937,
    QUEST_PORTAL_OF_THE_LEGION  = 5581
};

class go_demon_portal : public GameObjectScript
{
    public:
        go_demon_portal() : GameObjectScript("go_demon_portal") { }

        bool OnGossipHello(Player* player, GameObject* go)
        {
            if (player->GetQuestStatus(QUEST_PORTAL_OF_THE_LEGION) == QUEST_STATUS_INCOMPLETE && !go->FindNearestCreature(NPC_DEMON_GUARDIAN, 5.0f, true))
            {
                if (Creature* guardian = player->SummonCreature(NPC_DEMON_GUARDIAN, go->GetPositionX(), go->GetPositionY(), go->GetPositionZ(), 0.0f, TEMPSUMMON_DEAD_DESPAWN, 0))
                    guardian->AI()->AttackStart(player);
            }

            return true;
        }
};

void AddSC_desolace()
{
	// Ours
	new npc_cork_gizelton();

	// Theirs
    new npc_aged_dying_ancient_kodo();
    new npc_dalinda();
    new go_demon_portal();
}
